package com.itany.corejava.code13_jdbc;

import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;

/**
 * @author MissChen
 * @date 2024年03月22日 9:42
 */
public class Test06_保存返回主键 {
    public static void main(String[] args) throws DataAccessException {
        User user=new User("alice4","123","123","nanjing");
        int id=insertReturnPrimaryKey(user);
        System.out.println("主键值:"+id);
    }
    public static Integer insertReturnPrimaryKey(User user) throws DataAccessException {
        Connection conn=null;
        PreparedStatement ps=null;
        ResultSet rs=null;
        Integer id=null;
        try {
            conn=JDBCUtil.getConnection();
            String sql=new StringBuffer()
                    .append(" insert into ")
                    .append(" t_user ")
                    .append(" (username,password,phone,address) ")
                    .append(" values ")
                    .append(" (?,?,?,?) ")
                    .toString();
            //保存返回主键的功能
            // prepareStatement(sql,用于设置保存是否返回自动生成主键的标志)
            // RETURN_GENERATED_KEYS:返回自动生成的主键
            // NO_GENERATED_KEYS:不返回自动生成的主键
            ps=conn.prepareStatement(sql,PreparedStatement.RETURN_GENERATED_KEYS);
            ps.setString(1,user.getUsername());
            ps.setString(2,user.getPassword());
            ps.setString(3,user.getPhone());
            ps.setString(4,user.getAddress());
            //执行DML操作
            ps.executeUpdate();
            //当执行完DML操作后，在数据库中会生成对应的主键的值
            // 此时我们需要获取到生成后的主键的值
            // Statement中提供了获取自动生成的主键的方法
            // 其返回值是一个结果集
            rs=ps.getGeneratedKeys();

            while(rs.next()){
                //由于生成的主键是有数据库自动操作的
                // 对于开发者而言，数据库厂商为其定义列名我们不知道
                // 虽然开发者不知道列名，但是能够确定的是返回的是仅有一列
                // 因此可以通过索引号进行映射，直接将第一列的数据映射到对应的主键值
                id=rs.getInt(1);
            }
            return id;
        } catch (DataAccessException e) {
            throw e;
        } catch (SQLException e) {
            throw new DataAccessException("数据库访问失败了");
        }
    }
}
